import q3
import q1
import pickle
import matplotlib.pyplot as plt
import numpy as np

model = pickle.load(open('q3dingjia.pkl', 'rb'))
x=np.arange(0,0.3,0.001)
result=model.predict(x)**2
result=result.T[0]

fig = plt.figure()
ax = fig.add_subplot(111)
ax.scatter(q3.allDist,q1.allJiageA,c='blue')
ax.scatter(x,result,c='red')
plt.show()
